package org.example.demo2;

import javax.crypto.SecretKey;
import java.util.Base64;

public class Main {
    public static void main(String[] args) {
        try {
            // 生成密钥
//            SecretKey key = AESUtil.generateKey();
            String keyBase64="K3/KQrAQNomykFdy5/bgOQ==";
            SecretKey key = KeyUtil.getKeyFromBase64(keyBase64);
            // 输出密钥的 Base64 编码
            String encodedKey = Base64.getEncoder().encodeToString(key.getEncoded());
            System.out.println("生成的 AES 密钥（Base64 编码）： " + encodedKey);
            // 加密图片
            String inputFile = "demo2/copy.jpg";
            String encryptedFile = "demo2/encrypted_image.enc";
            AESUtil.encrypt(inputFile, encryptedFile, key);
            System.out.println("加密成功！");

            // 解密图片
            String decryptedFile = "demo2/decrypted_image.jpg";
            AESUtil.decrypt(encryptedFile, decryptedFile, key);
            System.out.println("解密成功！");

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
